package com.surrey.p2p.view;

import java.awt.Color;
import java.awt.Paint;
import java.util.ArrayList;

import org.apache.commons.collections15.Transformer;

public class EdgeColourTransformer implements Transformer<String, Paint> {

ArrayList<String> colored;	
	
	EdgeColourTransformer(ArrayList<String> colored){
		this.colored=colored;
	}
	
	
	@Override
	public Paint transform(String arg0) {
	  Color p= Color.LIGHT_GRAY;
		if(colored.contains(arg0)){
			p=Color.BLUE;
		}
	  return p;
	}

}
